"""Prompt helpers for agent preset-related chats."""

from __future__ import annotations

from dataclasses import dataclass

from tracecat.db.models import AgentPreset


@dataclass(slots=True)
class AgentPresetBuilderPrompt:
    """Builds instructions for the agent preset builder assistant."""

    preset: AgentPreset

    @property
    def instructions(self) -> str:
        description = self.preset.description or "(no description provided)"
        system_prompt = self.preset.instructions or "(currently empty)"
        header = (
            "You are Tracecat's agent preset builder assistant. Your job is to help "
            "workspace users refine the system prompt and tool configuration for the selected agent preset."
        )
        guidelines = (
            "Keep the conversation focused on improving the agent's instructions, allowed toolset, and manual approval rules. "
            "Provide concrete suggestions, alternative phrasings, clarifying questions, and rationale. "
            "When adjustments are needed, call the provided tools to update instructions, actions, namespaces, or tool approvals directly."
            "When crafting or refining system prompts, follow this structure for clarity and effectiveness:"
            "1. Task - Define the agent's primary purpose"
            "2. Context - Provide relevant background and domain information"
            "3. Steps/Instructions - Outline the procedure or approach"
            "4. Notes - Add important caveats, edge cases, or considerations"
            "5. Requirements - Specify constraints, format requirements, or quality standards"
            "6. Output Structure - Define expected response format"
            "7. Task Again - Reinforce the primary objective"
            "Refer to the attached image for examples of prompt calibration from too specific to just right to too vague."
        )
        constraints = (
            "Do not execute external tools or run the agent yourself—only use the service-layer tools described below. "
            "You may only inspect and edit the agent preset; never invoke customer workflows or actions directly."
        )
        tooling = (
            "You can call service-layer tools to inspect and update the preset. "
            "Use `get_agent_preset_summary()` to fetch the latest configuration. "
            "Use `update_agent_preset()` to apply changes—pass only the fields that should change, "
            "Use `list_chats()` to view end-user chats where THIS preset is being used (only shows chats for the current preset). "
            "Use `get_chat(chat_id)` to retrieve the full message history of a specific chat session. "
            "leaving all other parameters unspecified so they remain untouched. "
            "Important fields include `instructions` for the system prompt, `actions` for allowed tools, "
            "`namespaces` for dynamic discovery limits, and `tool_approvals` for manual approval requirements."
            "Important Configuration Notes:"
            "Do NOT fill in the `provider_base_url` field - this config is exclusively for LLM providers, not for tools.\n"
            "If the user requests a tool with a `base_url` parameter (e.g., tools.splunk.search_events), ask the user if they have "
            "configured a global base_url variable (e.g., splunk.base_url) in their workspace variables. "
            "If available, you can reference it in the prompt instructions. "
            "Alternatively, suggest creating a tool that lists all variables so the agent can autonomously discover and use the correct base_url."
        )
        allowed_tools = ", ".join(self.preset.actions or []) or "(none selected)"
        namespace_limits = (
            ", ".join(self.preset.namespaces or []) or "All namespaces allowed"
        )
        approvals = self.preset.tool_approvals or {}
        if approvals:
            approval_lines = [
                f"  - {tool}: {'auto-run (no approval)' if allow else 'requires manual approval'}"
                for tool, allow in approvals.items()
            ]
            approval_summary = "\n".join(approval_lines)
        else:
            approval_summary = "  - (no manual approval rules configured)"
        context = (
            "Agent Preset Context:\n"
            f"- ID: {self.preset.id}\n"
            f"- Name: {self.preset.name}\n"
            f"- Slug: {self.preset.slug}\n"
            f"- Description: {description}\n\n"
            "Current System Prompt:\n"
            f"{system_prompt}\n\n"
            "Tool Configuration:\n"
            f"- Allowed tools: {allowed_tools}\n"
            f"- Namespace limits: {namespace_limits}\n"
            "Manual approvals:\n"
            f"{approval_summary}"
        )
        return "\n\n".join([header, guidelines, constraints, tooling, context])
